package com.lzjn.code.web.util;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipUtil {

	private static Logger log = LoggerFactory.getLogger(ZipUtil.class);
	
	public static void createZip(String sourcePath, String zipPath) {
		
	    FileOutputStream fos = null;
	    ZipOutputStream zos = null;
	    
	    try {
	    	
	        fos = new FileOutputStream(zipPath);
	        zos = new ZipOutputStream(fos);
	        writeZip(new File(sourcePath), "", zos);
	    } 
	    catch (FileNotFoundException e) {
	    	
	    	log.error("创建ZIP文件失败", e);
	    } 
	    finally {
	    	
	        try {
	            if (zos != null) {
	            	
	                zos.close();
	            }
	        } catch (IOException e) {
	        	
	            log.error("创建ZIP文件失败", e);
	        }
	
	    }
	}

	private static void writeZip(File file, String parentPath, ZipOutputStream zos) {
		
	    if(file.exists()){
	    	
	        if(file.isDirectory()) {//处理文件夹
	
	            parentPath+=file.getName()+File.separator;
	            File [] files=file.listFiles();
	            for(File f:files){
	                writeZip(f, parentPath, zos);
	            }
	        }
	        else {
	            FileInputStream fis=null;
	            DataInputStream dis=null;
	            try {
	            	
	                fis=new FileInputStream(file);
	                dis=new DataInputStream(new BufferedInputStream(fis));
	                ZipEntry ze = new ZipEntry(parentPath + file.getName());
	                zos.putNextEntry(ze);
	                byte [] content=new byte[1024];
	                int len;
	                while((len=fis.read(content))!=-1){
	                	
	                    zos.write(content,0,len);
	                    zos.flush();
	                }
	            } catch (FileNotFoundException e) {
	            	
	                log.error("创建ZIP文件失败",e);
	            } catch (IOException e) {
	            	
	                log.error("创建ZIP文件失败",e);
	            }finally{
	            	
	                try {
	                    if(dis!=null){
	                    	
	                        dis.close();
	                    }
	                }catch(IOException e){
	                	
	                    log.error("创建ZIP文件失败",e);
	                }
	            }
	        }
	    }
	}   

}